import socket


# 聊天器主要功能
# 1、发送消息
# 2、接收消息
# 3、退出消息

def send_msg(udp_socket):
    """发送消息的方法"""
    # 输入接收方的ip地址
    ipaddr = input("请输入接收方地址：\n")
    if len(ipaddr) == 0:
        ipaddr = "127.0.0.1"
        print("默认设置为：%s" % ipaddr)

    port = input("请输入接收方端口号：\n")
    if len(port) == 0:
        port = "6666"
        print("默认设置为：%s" % port)

    content = input("请输入发生内容：\n")

    udp_socket.sendto(content.encode(), (ipaddr, port))


def recv_msg(udp_socket):
    # 1、接收数据
    recv_data = udp_socket.recvfrom(1024)
    # 2、把接收到的数据解码并且显示出来
    re_text = recv_data[0].decode()
    print("接收到消息为：%s" % re_text)
    ip_port = recv_data[1]
    # 3、打印显示发送⽅的IP和端⼝信息
    print(ip_port)


if __name__ == '__main__':
    # 创建套接字
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    # 绑定端⼝为 6666
    udp_socket.bind(("", 6666))
    while True:
        # 给出提示，显示聊天器的主要功能
        print("*************************")
        print("**** 1.发送消息 ****")
        print("**** 2.接收消息 ****")
        print("**** 3.退出系统 ****")
        print("*************************")
        # 1、提示⽤户选择功能
        num = int(input("请选择功能:\n"))
        # 2、判断⽤户选择
        # 如果选择1,调⽤发送消息的函数
        if num == 1:
            # print("您选择了发送消息")
            send_msg(udp_socket)
        # 如果选择2,调⽤接收消息的函数
        elif num == 2:
            recv_msg(udp_socket)
        # 如果选择3,退出程序执⾏
        else:
            print("程序正在退出...")
            break

        print("程序已退出！～")

    # 关闭套接字
    udp_socket.close()
